python solve formula